LESS Is More: Mutual-Stability Sampling for Diffusion Language Models
LESS 即更多:扩散语言模型的互稳定采样
🧭 扩散语言模型推理加速
MBZUAI · Ecole Polytechnique
提出训练无关的自适应采样器 LESS,通过联合稳定性规则在线决定 token 揭示时机,以 72.1% 更少步骤保持精度。
📌 工作参考价值(4 点)
- 若业务中尝试将扩散模型用于生成式推荐(如 Semantic ID 生成),LESS 可嵌入推理流程,自适应提前终止不稳定的 token,直接减少 Transformer 前向次数并降低延迟。
- 其 top-K JS 散度稳定性判据与 top-1 置信度/持久性规则可以独立迁移到其他 token 级早停(early exiting)场景,比如在序列生成中按位置动态分配计算量。
- 该采样器无需训练、模型无关,可作为即插即用模块快速集成到现有扩散 LLM 部署中,无需重新训练或修改模型权重。
- 本文主要学术贡献在于扩散 LM 的高效采样,若业务尚未使用扩散模型,直接可借鉴点有限;但在 Agent 系统里若用扩散模型做规划或工具调用生成,LESS 可加速推理。